import argparse

def conf_load():
    se = argparse.ArgumentParser(description="测试函数参数设置")
    #TODO
    se.add_argument("--model", type=str, default="your_model_path",
                        help="模型的路径，例如 'experiment/models/CodeLlama-7b-hf'")
    se.add_argument("--adapter", type=str, default="../checkpoints/mix_cl_a50_e1_l0.0005/best_checkpoint",
                        help="适配器的路径，保存的模型参数位置")
    se.add_argument("--device", type=str, default="cuda:0",
                        help="运行设备，例如 'cuda:0' 或 'cpu'")
    se.add_argument("--in_way", type=str, default="../data/test_set_B/Q_B_without_answer.jsonl",
                        help="输入文件路径，默认为 '../data/test_set_B/Q_B_without_answer.jsonl'")
    se.add_argument("--outfile", type=str, default="final_res",
                        help="输出文件路径，默认为 'final_res'")
    se.add_argument("--type", type=str, default="bfloat16",
                        help="指定 Torch 数据类型，例如 'bfloat16' 或 'float32'")
    se.add_argument("--sed", type=int, default=2618,
                        help="随机种子")

    conf = se.parse_args()
    return conf
